import torch
import os
os.environ["HF_ENDPOINT"]="https://hf-mirror.com"

from transformers import AutoTokenizer, AutoModelForCausalLM
model = AutoModelForCausalLM.from_pretrained("FlagAlpha/Llama2-Chinese-7b-Chat")